package com.cy.ragbase.dto;

import lombok.Data;

@Data
public class RagPromptRequest {
    /**
     * 用户提出的问题
     */
    private String question;

    /**
     * 搜索返回的文档数量，默认5个
     */
    private int topK = 5;

    /**
     * 相似度阈值，默认0.7
     */
    private float threshold = 0.7f;

    /**
     * 提示词模板类型，默认为代码生成
     */
    private PromptType promptType = PromptType.CODE_GENERATION;

    public enum PromptType {
        CODE_GENERATION,    // 代码生成
        GENERAL_QA,        // 通用问答
        DOCUMENTATION      // 文档说明
    }
}